This PR fixes bug under is_safetensors_compatible function: when variant and safetensor file does not match, this function should return False. For example, in this model: stable-diffusion-v1-5/stable-diffusion-inpainting, there is file diffusion_pytorch_model.fp16.safetensors, but when variant is None, it should return False instead of True

@a-r-r-o-w @DN6 , pls help review, thx.

@bot /style

Style fixes have been applied. View the workflow run here.

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

Thanks @kaixuanliu! I think we would need to add the variant check when there are no component folders as well.

    if not components:
        return any(".safetensors" in filename for filename in filenames)

And I think the changes will break the current is_safetensors_compatible tests here

@DN6 , Hi, thanks for your advice. Have updated the code and test cases

@bot /style

Style fixes have been applied. View the workflow run here.

@DN6 Hi, I have checked all the CI tests offline. Can you help trigger the CI workflow again?